# import netron
# '''
# view the model by netron
#
# netron have 2 version
# 1、 netron.exe:Wq
# 2、 import netron
#     netron.start(model_path)
# '''
# netron.start("weights/netron.pth")

# from torchstat import stat
from torchvision.models import resnet18

model=resnet18().to("cuda")
# stat(model,(3,224,224))


from torchsummary import summary

summary(model,(3,224,224))


